#!/bin/bash

# 第一个参数是 GPU 编号，后面的参数是传递给 docker run 的其他参数
GPU=$1 
shift
DIR=/Repmet/models 

# 输出 GPU 编号和容器目录
echo "GPU = $GPU"
echo "Container Directory: $DIR"

# 使用 docker run 命令启动容器，指定 GPU，容器名，卷挂载和其他参数
# 注意：这里假设您想要将宿主机的 $HOME 目录挂载到容器的 $DIR 目录
# 并且您已经有一个名为 repmet 的镜像可用
docker run --gpus "device=$GPU" --rm --name repmet_training -v "$HOME":"$DIR" -it repmet "$@"
